Fixed beta code

Signed-off-by: David Evans <dgevans42@gmail.com>