Small fix of camera. TODO: Move camera update code