Bugfixing

Fixer a small bug