Idea/Fix

Just a quick fix :P