Fixed wrong code