Slight Problem with update 0.0.6

Fixed slight problem with update to 0.0.6