more fixes after merge

Signed-off-by: Jake Whatley <jacob.whatley91@gmail.com>