Slight bugfix