Fixed minor overlooked change from last merge