Small bug fixxed