Fix one little bug