Bug fixed

Fixed lil bug pointed out by @jroedel