Made fix suggested by rxu