Repaired stupid mistake