Fixes oversight in code.