Another small fix, originally worked on this in an older codebase (my bad)