added a check

Signed-off-by: ShadowChild <shadychild12@gmail.com>