Fixed little bug