Fix provided by Frankenstien, thank you!

Fix provided by Frankenstien, thank you!