Adjust some changes from the last commit

Signed-off-by: Steven Downer <grinch@outlook.com>