fix major issue introduced in last commit