Slight fix for last commit