#6 Fix

Fix a stupid bug