Fix other little bug