Build fix (added proper casting)