Fix code error

Signed-off-by: Matthew Ell <ell.matthew@gmail.com>